package com.wt.admin.domain.dto.ai;


import com.wt.admin.domain.entity.ai.AgentsInfoEntity;
import lombok.Data;

@Data
public class ChatDTO {

    private Integer agentsId;

    private String message;

    private Integer contentId;

    private AgentsInfoEntity agents;

    private ChatConfig chatConfig = new ChatConfig();

    @Data
    public static class ChatConfig{
        // 相似文档的最大数量
        private Integer topK = 5;
        // 介于 0 到 1 之间的双精度值，其中值越接近 1 表示相似度越高
        private Double similarityThreshold = 0.5;
        // 上下文长度
        private Integer contextLength = 100;
        // 提示词
        private String prompt = "";
        // 是否启用协同
        private Boolean enableSynergism = false;
    }
}
